import numpy as np


# 保存模型
def save_model(weights, filename='model.npy'):
    np.save(filename, weights)
    print(f"Model saved as {filename}")


# 加载模型
def load_model(filename='model.npy'):
    weights = np.load(filename)
    print(f"Model loaded from {filename}")
    return weights
